package com.star.common.kit.http;

import com.star.common.core.contract.enums.ResultCodeEnum;
import com.star.common.core.utils.BizAssert;
import com.star.common.kit.contract.exception.HttpClientException;
import org.apache.commons.lang3.StringUtils;
import org.apache.http.NameValuePair;
import org.apache.http.client.entity.UrlEncodedFormEntity;
import org.apache.http.client.methods.CloseableHttpResponse;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.client.utils.URIBuilder;
import org.apache.http.entity.ContentType;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.message.BasicNameValuePair;
import org.apache.http.util.EntityUtils;

import java.io.IOException;
import java.net.URI;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

/**
 * Http 工具类
 *
 * @author xtp
 * @date 2023/11/27
 */
public class HttpUtil {

    private HttpUtil() {
    }

    /**
     * 带参数的get请求
     *
     * @param url   请求地址
     * @param param 请求参数
     * @return String
     */
    public static String doGet(String url, Map<String, String> param) {
        // 创建Httpclient对象
        BizAssert.isFalseWarn(StringUtils.isBlank(url) || url.isEmpty(), ResultCodeEnum.URL_IS_NULL);
        CloseableHttpClient httpclient = HttpClients.createDefault();
        String resultString = "";
        CloseableHttpResponse response = null;
        try {
            // 创建uri
            URIBuilder builder = new URIBuilder(url);
            if (param != null) {
                for (Map.Entry<String, String> entry : param.entrySet()) {
                    String key = entry.getKey();
                    String value = entry.getValue();
                    builder.addParameter(key, value);
                }
            }
            URI uri = builder.build();
            // 创建http GET请求
            HttpGet httpGet = new HttpGet(uri);
            // 执行请求
            response = httpclient.execute(httpGet);
            // 判断返回状态是否为200
            if (response.getStatusLine().getStatusCode() == 200) {
                resultString = EntityUtils.toString(response.getEntity(), StandardCharsets.UTF_8);
            }
        } catch (Exception e) {
            throw new HttpClientException(e.getMessage(), e);
        } finally {
            try {
                if (response != null) {
                    response.close();
                }
                httpclient.close();
            } catch (IOException e) {
                throw new HttpClientException(e.getMessage(), e);
            }
        }
        return resultString;
    }

    /**
     * 不带参数的get请求
     *
     * @param url 请求地址
     */
    public static String doGet(String url) {
        return doGet(url, null);
    }


    /**
     * 带header参数的get请求
     *
     * @param url     请求地址
     * @param headers 请求头参数
     * @return String 请求结果
     */
    public static String doGetByHeaders(String url, Map<String, String> headers) {
        // 创建Httpclient对象
        if (url == null || url.isEmpty()) {
            return null;
        }
        CloseableHttpClient httpclient = HttpClients.createDefault();

        String resultString = "";
        CloseableHttpResponse response = null;
        try {
            // 创建uri
            URIBuilder builder = new URIBuilder(url);
            URI uri = builder.build();
            // 创建http GET请求
            HttpGet httpGet = new HttpGet(uri);
            if (headers != null) {
                for (Map.Entry<String, String> entry : headers.entrySet()) {
                    String key = entry.getKey();
                    String value = entry.getValue();
                    httpGet.addHeader(key, value);
                }
            }
            // 执行请求
            response = httpclient.execute(httpGet);
            // 判断返回状态是否为200
            if (response.getStatusLine().getStatusCode() == 200) {
                resultString = EntityUtils.toString(response.getEntity(), StandardCharsets.UTF_8);
            }
        } catch (Exception e) {
            throw new HttpClientException(e.getMessage(), e);
        } finally {
            try {
                if (response != null) {
                    response.close();
                }
                httpclient.close();
            } catch (IOException e) {
                throw new HttpClientException(e.getMessage(), e);
            }
        }
        return resultString;
    }

    /**
     * 带header参数的get请求
     *
     * @param url    请求地址
     * @param orgKey Org-Key请求头参数
     * @return String 请求结果
     */
    public static String doGetByHeader(String url, String orgKey) {
        // 创建Httpclient对象
        BizAssert.isFalseWarn(StringUtils.isBlank(url) || url.isEmpty(), ResultCodeEnum.URL_IS_NULL);
        CloseableHttpClient httpclient = HttpClients.createDefault();

        String resultString = "";
        CloseableHttpResponse response = null;
        try {
            // 创建uri
            URIBuilder builder = new URIBuilder(url);
            URI uri = builder.build();
            // 创建http GET请求
            HttpGet httpGet = new HttpGet(uri);
            httpGet.addHeader("Org-Key", orgKey);
            // 执行请求
            response = httpclient.execute(httpGet);
            // 判断返回状态是否为200
            if (response.getStatusLine().getStatusCode() == 200) {
                resultString = EntityUtils.toString(response.getEntity(), StandardCharsets.UTF_8);
            }
        } catch (Exception e) {
            throw new HttpClientException(e.getMessage(), e);
        } finally {
            try {
                if (response != null) {
                    response.close();
                }
                httpclient.close();
            } catch (IOException e) {
                throw new HttpClientException(e.getMessage(), e);
            }
        }
        return resultString;
    }

    /**
     * 带参数的post请求
     *
     * @param url   请求地址
     * @param param 请求参数
     * @return String 请求结果
     */
    public static String doPost(String url, Map<String, String> param) {
        // 创建Httpclient对象
        CloseableHttpClient httpClient = HttpClients.createDefault();
        CloseableHttpResponse response = null;
        String resultString = "";
        try {
            // 创建Http Post请求
            HttpPost httpPost = new HttpPost(url);
            httpPost.addHeader("Content-Type", "application/json; charset=utf-8");
            httpPost.addHeader("X-Api-Sign-Version", "2.0.0");
            // 创建参数列表
            if (param != null) {
                List<NameValuePair> paramList = new ArrayList<>();
                for (Map.Entry<String, String> entry : param.entrySet()) {
                    String key = entry.getKey();
                    String value = entry.getValue();
                    paramList.add(new BasicNameValuePair(key, value));
                }
                // 模拟表单
                UrlEncodedFormEntity entity = new UrlEncodedFormEntity(paramList);
                httpPost.setEntity(entity);
            }
            // 执行http请求
            response = httpClient.execute(httpPost);
            resultString = EntityUtils.toString(response.getEntity(), StandardCharsets.UTF_8);
        } catch (Exception e) {
            throw new HttpClientException(e.getMessage(), e);
        } finally {
            try {
                response.close();
            } catch (IOException e) {
                throw new HttpClientException(e.getMessage(), e);
            }
        }
        return resultString;
    }

    /**
     * 不带参数的post请求
     *
     * @param url 请求地址
     * @return String 请求结果
     */
    public static String doPost(String url) {
        return doPost(url, null);
    }

    /**
     * 传送json类型的post请求
     *
     * @param url  请求地址
     * @param json 请求参数
     * @return String
     */
    public static String doPostJson(String url, String json, String tokenName, String token) {
        // 创建Httpclient对象
        CloseableHttpClient httpClient = HttpClients.createDefault();
        CloseableHttpResponse response = null;
        String resultString = "";
        try {
            // 创建Http Post请求
            HttpPost httpPost = new HttpPost(url);
            httpPost.addHeader(tokenName, token);
            // 创建请求内容
            StringEntity entity = new StringEntity(json, ContentType.APPLICATION_JSON);
            httpPost.setEntity(entity);
            // 执行http请求
            response = httpClient.execute(httpPost);
            resultString = EntityUtils.toString(response.getEntity(), StandardCharsets.UTF_8);
        } catch (Exception e) {
            throw new HttpClientException(e.getMessage(), e);
        } finally {
            try {
                response.close();
            } catch (IOException e) {
                throw new HttpClientException(e.getMessage(), e);
            }
        }
        return resultString;
    }


}
